package com.wei.flow;

import java.util.HashMap;
import java.util.Map;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;

public class FlowPartitioner extends Partitioner<Text, FlowBean>{
	static Map map=new HashMap<>();
	static{
		map.put("135", 1);
		map.put("136", 2);
		map.put("137", 3);
		map.put("138", 4);
		map.put("139", 5);
	}
	@Override
	public int getPartition(Text phone, FlowBean arg1, int arg2) {
		Integer code=(int) map.get(phone.toString().substring(0, 3));
		return code == null ? 5 : code;
	}

	

}
